富岳およびSummitにおける核融合プラズマ流体解析の高速化

井戸村 泰宏; 伊奈 拓也*; Ali, Y.*; 今村 俊幸*

第34回数値流体力学シンポジウム講演論文集(インターネット), 6 Pages, 2020/12

ジャイロ運動論的トロイダル5次元full-fオイラーコードGT5Dにおける半陰解法差分計算用に新しいFP16(半精度)前処理付き省通信クリロフソルバを開発した。このソルバでは、大域的集団通信のボトルネックを省通信クリロフ部分空間法によって解決し、さらに収束特性を向上するFP16前処理によって袖通信を削減した。FP16前処理は演算子の物理特性に基づいて設計し、A64FXで新たにサポートされたFP16SIMD処理を用いた実装した。このソルバをGPUにも移植し、約1,000億格子のITER規模計算の性能を富岳(A64FX)とSummit(V100)で測定した。従来の非省通信型ソルバに比べて、新しいソルバはGT5Dを倍加速し、富岳とSummitの両方で5,760CPU/GPUまで良好な強スケーリングが得られた。